package com.tyf.common.util;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.Date;

/**
 * @author: QiangLiu
 * @create: 2017-04-21 上午9:58
 * @version: 1.0
 **/
public class ImgCompress {
    private static final Logger logger = LoggerFactory.getLogger(ImgCompress.class);
    private Image img;
    private int width;
    private int height;


    public ImgCompress(InputStream inputStream) throws IOException {
        img = ImageIO.read(inputStream);      // 构造Image对象
        width = img.getWidth(null);    // 得到源图宽
        height = img.getHeight(null);  // 得到源图长
    }

    public ImgCompress(String fileName) throws IOException {
        File file = new File(fileName);// 读入文件
        img = ImageIO.read(file);      // 构造Image对象
        width = img.getWidth(null);    // 得到源图宽
        height = img.getHeight(null);  // 得到源图长
    }

    /**
     * 按照宽度还是高度进行压缩
     *
     * @param w int 最大宽度
     * @param h int 最大高度
     */
    public ByteArrayOutputStream resizeFix(int w, int h) throws IOException {
        ByteArrayOutputStream out = null;
        if (width / height > w / h) {
            out = resizeByWidth(w);
        } else {
            out = resizeByHeight(h);
        }
        return out;
    }

    /**
     * 以宽度为基准，等比例放缩图片
     *
     * @param w int 新宽度
     */
    public ByteArrayOutputStream resizeByWidth(int w) throws IOException {
        int h = (int) (height * w / width);
        return resize(w, h);
    }

    /**
     * 以高度为基准，等比例缩放图片
     *
     * @param h int 新高度
     */
    public ByteArrayOutputStream resizeByHeight(int h) throws IOException {
        int w = (int) (width * h / height);
        return resize(w, h);
    }

    /**
     * 强制压缩/放大图片到固定的大小
     *
     * @param w int 新宽度
     * @param h int 新高度
     */
    public ByteArrayOutputStream resize(int w, int h) throws IOException {
        // SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
        BufferedImage image = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
        image.getGraphics().drawImage(img, 0, 0, w, h, null); // 绘制缩小后的图

        ByteArrayOutputStream out = new ByteArrayOutputStream(); // 输出到文件流

        ImageIO.write(image, "jpeg", out);
        return out;
    }

    /**
     * 强制压缩/放大图片到固定的大小
     *
     * @param w int 新宽度
     * @param h int 新高度
     */
    public void resize(String pathName, int w, int h) throws IOException {
        // SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
        BufferedImage image = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
        image.getGraphics().drawImage(img, 0, 0, w, h, null); // 绘制缩小后的图
        File destFile = new File(pathName);
        FileOutputStream out = new FileOutputStream(destFile); // 输出到文件流

        ImageIO.write(image, "jpeg", out);
        IOUtils.closeQuietly(out);
    }

    /**
     * out 流转换成input 流
     *
     * @param out
     * @return
     */
    public static ByteArrayInputStream parse(OutputStream out) {
        ByteArrayOutputStream baos = null;
        if (out instanceof ByteArrayOutputStream) {

            baos = (ByteArrayOutputStream) out;
            byte[] bytes = baos.toByteArray();
            ByteArrayInputStream swapStream = new ByteArrayInputStream(bytes);

            IOUtils.closeQuietly(baos);

            return swapStream;
        }
        return null;
    }


    public static void main(String[] args) throws IOException {
        System.out.println("开始：" + new Date().toLocaleString());
        ImgCompress imgCom = new ImgCompress("/Users/qiangliu/Downloads/0009.png");
        String path = "/Users/qiangliu/Downloads/0008.png";
        imgCom.resize(path, 400, 400);

        System.out.println("结束：" + new Date().toLocaleString());
    }
}
